load("@rules_cc//cc:defs.bzl", "cc_library")
load("@rules_cuda//cuda:defs.bzl", "cuda_library")

package(default_visibility = ["//visibility:public"])

cuda_library(
    name = "gpu",
    srcs = [
        "air_detector/gpu.cu",
    ],
    hdrs = [
        "air_detector/gpu.h",
    ],
    deps = [
        "@com_github_google_glog//:glog",
        "@opencv//:opencv",
        "@cuda//:cuda",
    ],
)

cc_library(
    name = "detector_plugin_adapter",
    srcs = glob([
        "*.cc",
        "**/*.cc",
    ]),
    hdrs = glob([
        "*.h",
        "**/*.h",
    ]),
    deps = [
        "//base/plugin:base_plugin",
        "//base/io:base_io",
        "//base/blob:base_blob",
        "//base/cv_image:base_cv_image",
        "//base/common:base_common",
        "//air_service/modules/perception-camera/algorithm/interface:camera_algorithm_interface",
        "//air_service/modules/perception-camera/proto:detector_param_cc_pb",
        "@cuda//:cuda",
        "@paddle//:paddle",
        "@tensorrt//:tensorrt",
        "gpu"
    ],
)

